package com.jc.oss.util;

import com.jc.oss.entity.fileInfo;
import io.minio.*;
import io.minio.errors.*;
import io.minio.messages.Bucket;
import io.minio.messages.Item;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

import javax.annotation.Resource;
import java.io.BufferedInputStream;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.security.InvalidKeyException;
import java.security.NoSuchAlgorithmException;
import java.util.LinkedList;
import java.util.List;
import java.util.stream.Collectors;

/**
 * @Author gongjiawei
 * @Date 2024/8/30 19:58
 * @注释
 */
@Component
public class MinioUtil {
    @Resource
    private MinioClient minioClient;

    /*创建bucket桶
     * */
    public void createBucket(String bucket) throws Exception {
        boolean exists = minioClient.bucketExists(BucketExistsArgs.builder().bucket(bucket).build());
        if (!exists) {
            minioClient.makeBucket(MakeBucketArgs.builder().bucket(bucket).build());
        }
    }

    /*
     *
     * 上传文件
     *
     */
//    public void uploadFile(InputStream inputStream, String bucket, String objectName) throws Exception {
//        minioClient.putObject(
//                PutObjectArgs.builder()
//                        .bucket(bucket)
//                        .object(objectName)
//                        .stream(inputStream, -1, Integer.MAX_VALUE)
//                .build());
//    }
    public void uploadFile(InputStream inputStream, String bucket, String objectName) throws Exception {
        minioClient.putObject(PutObjectArgs.builder().bucket(bucket).object(objectName)
                .stream(inputStream, -1, 5242889L).build());
    }

    /*
    * 列出所有桶
    *
    * */

    public List<String> getAllBucket()throws Exception{
        List<Bucket>buckets=minioClient.listBuckets();
        return buckets.stream().map(Bucket::name).collect(Collectors.toList());

    }

    /*
     * 列出当前桶及文件
     *
     * */

    public List<fileInfo> getAllFile(String bucket)throws Exception{
        Iterable<Result<Item>> results = minioClient.
                listObjects(ListObjectsArgs.builder().bucket(bucket).build());
        List<fileInfo> fileInfoList=new LinkedList<>();
        for (Result<Item> result : results) {
            fileInfo fileInfo = new fileInfo();
            Item item = result.get();
            fileInfo.setFileName(item.objectName());
            fileInfo.setEtag(item.etag());
            fileInfo.setDirectoryFlag(item.isDir());
            fileInfoList.add(fileInfo);
        }
        return fileInfoList;

    }


    /*
     * 下载文件
     *
     * */
    public InputStream download(String bucket,String objectName) throws Exception{
        return minioClient.getObject(GetObjectArgs.builder().bucket(bucket).object(objectName).build());
    }

    /*
     * 删除桶
     *
     * */
    public void delete(String bucket) throws Exception{
         minioClient.removeBucket(RemoveBucketArgs.builder().bucket(bucket).build());
    }
    /*
     * 删除文件
     *
     * */
    public void delete(String bucket,String objectName) throws Exception{
        minioClient.removeObject(RemoveObjectArgs.builder().bucket(bucket).object(objectName).build());
    }

}
